package com.example.producer.config;

import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.support.converter.Jackson2JsonMessageConverter;
import org.springframework.amqp.support.converter.MessageConverter;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.annotation.PostConstruct;
import javax.annotation.Resource;

/**
 * @description:
 * @program: demos
 * @author: dove
 * @date: 2020-12-09 00:35
 * 采用json格式的序列化器，mq中的消息存放的是json格式的数据，而不是二进制数据
 **/

@Configuration
public class MyRabbitMQConfig {

    @Resource
    private RabbitTemplate rabbitTemplate;

    /**
     * 使用json序列化机制，进行消息转换。
     * 如果只需要一个MessageConverter，则无需注入RabbitTemplate
     * @return
     */
    @Bean
    public MessageConverter getMessageConverter(){
        return new Jackson2JsonMessageConverter();
    }

    /**
     * 定制RabbitTemplate
     *  1.服务器收到消息就回调
     *      1.配置spring.rabbitmq.publisher-confirms=true
     *      2.设置确认回调
     *  2.消息未正确抵达队列进行回调
     *      1.spring.rabbitmq.publisher-returns=true
     *        spring.rabbitmq.template.mandatory=true
     *      2.设置未抵达回调ReturnCallback（只要消息没有投递给指定的队列，就触发这个失败回调）
     *  3.消费端确认机制（保证每一个消息被正确消费，此时broker才可以删除这个消息）
     *      1.默认自动确认，只要消息接收到，客户端就会自动确认这个消息，服务端就会移除这个消息。
     *          自动确认的问题：
     *              消费者收到很多消息，由于服务器宕机，导致未处理的消息自动确认，服务端移除了这些消息，造成消息丢失。
     *          解决方式：手动确认消息. 未手动确认的消息都是unacked模式
     *      2.如何确认：手动ack
     *
     * @PostConstruct: MyRabbitConfig对象创建完成以后，执行这个方法
     */
    @PostConstruct
    public void initRabbitTemplate(){
        // 设置确认回调
        rabbitTemplate.setConfirmCallback(new RabbitTemplate.ConfirmCallback() {
            /**
             * @param correlationData: 当前消息的唯一关联数据（这个是消息的唯一ID）
             *                         CorrelationData配合数据库本地事务表，来确保消息的可靠性
             *                         消息发送之前先记录消息的CorrelationDataId,消息消费后去数据库修改该条消息的状态，表示该消息已经收到并消费
             *                         未消费的消息可以再次消费
             * @param ack：消息是否成功，消息抵达服务器就会返回true
             *
             * @param cause：失败的原因
             */
            @Override
            public void confirm(CorrelationData correlationData, boolean ack, String cause) {
                System.out.println("ConfirmCallback--->correlationData:["+correlationData+"],ack:["+ack+"],cause:["+cause+"]");
            }
        });

        // 设置消息抵达队列的确认回调
        rabbitTemplate.setReturnCallback(new RabbitTemplate.ReturnCallback() {
            /**
             * 只要消息没有投递给指定的队列，就触发这个失败回调
             * @param message    投递失败的消息详细信息
             * @param replyCode  回复的状态码
             * @param replyText  回复的文本内容: NO_ROUTE
             * @param exchange   当时这个消息发送给那个交换机
             * @param routingKey 使用的routingKey是什么
             */
            @Override
            public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
                System.out.println("ReturnCallback---->message:["+message+"],replyCode:["+replyCode+"],replyText:["+replyText+"],exchange:["+exchange+"],routingKey:["+routingKey+"]");
            }
        });
    }

}
